#include <iostream>
#include <string>
#include <unistd.h>
#include <signal.h>
#include "../comm/httplib.h"
#include "oj_control.hpp"

using namespace httplib;
using namespace ns_control;
static Control* ctrlptr = nullptr;

void Recovery(int signo)
{
    ctrlptr->RecoveryMachine();
}
int main()
{
    signal(SIGQUIT, Recovery); // 通过信号一件上线所有主机
    Server svr; //httplib
    Control ctrl; // Control 类
    ctrlptr = &ctrl;
    // lambda 表达式中要使用外部变量，需要在 [] 里引用

    // 功能路由
    // 请求所有题目
    svr.Get("/all_questions", [&ctrl](const Request& req, Response &resp){
        // 返回一张有题目的网页 
        std::string html;
        ctrl.AllQuestions(&html); // 获取题目并渲染
        resp.set_content(html, "text/html;charset=utf-8"); // 将获取到的网页返回 
    });

    // 根据用户所需题目编号, 获取题目内容
    svr.Get(R"(/question/(\d+))", [&ctrl](const Request& req, Response &resp){
        std::string number = req.matches[1];
        std::string html;
        ctrl.Question(number, &html);
        resp.set_content(html, "text/html;charset=utf-8");
    });

    // 用户提交代码后，实现判题功能 
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request& req, Response &resp){
        std::string number = req.matches[1];
        std::string result_json;
        ctrl.Judge(number, req.body, &result_json);
        resp.set_content(result_json, "application/json;charset=utf-8");
    });

    svr.set_base_dir("./wwwroot");
    daemon(true, false);
    svr.listen("0.0.0.0", 3306); // 启动服务
    return 0;
}